# 企业提示词构造
def get_industry_trans_func(attr_name, is_industry_prompt):
    """
    整个表格的各个属性信息，拼凑出企业的相关信息，但不包含提示词内容。
    相关信息保存到 attr_name 属性中。
    map func of dataset:
        dataset = dataset.map(build_prompt)
    """

    def _func(item):
        col_name = [
            "企业名称",
            "经营范围",
            "一级行业分类",
            "二级行业分类",
            "三级行业分类",
        ]
        industry_info = []
        pair_prompt = "* {key}: {value}\n"

        for col in col_name:
            industry_info.append(pair_prompt.format(key=col, value=item[col]))

        item[attr_name] = is_industry_prompt.format(
            industry_info="\n".join(industry_info)
        )
        return item

    return _func


if __name__ == "__main__":
    # get_industry_trans_func()
    pass
